/*
 * This file is part of aion-engine <aion-engine.com>
 *
 * aion-engine is private software: you can redistribute it and or modify
 * it under the terms of the GNU Lesser Public License as published by
 * the Private Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * aion-engine is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser Public License for more details.
 *
 * You should have received a copy of the GNU Lesser Public License
 * along with aion-engine.  If not, see <http://www.gnu.org/licenses/>.
 */
package com.aionengine.gameserver.dataholders;

import com.aionengine.gameserver.model.templates.materials.MaterialSkill;
import com.aionengine.gameserver.model.templates.materials.MaterialTemplate;

import javax.xml.bind.Unmarshaller;
import javax.xml.bind.annotation.*;
import java.util.*;

import static ch.lambdaj.Lambda.extract;
import static ch.lambdaj.Lambda.on;

/**
 * @author Rolandas
 */

@XmlAccessorType(XmlAccessType.FIELD)
@XmlType(name = "", propOrder = {"materialTemplates"})
@XmlRootElement(name = "material_templates")
public class MaterialData {

    @XmlElement(name = "material")
    protected List<MaterialTemplate> materialTemplates;

    @XmlTransient
    Map<Integer, MaterialTemplate> materialsById = new HashMap<Integer, MaterialTemplate>();

    @XmlTransient
    Set<Integer> skillIds = new HashSet<Integer>();

    void afterUnmarshal(Unmarshaller u, Object parent) {
        if (materialTemplates == null)
            return;

        for (MaterialTemplate template : materialTemplates) {
            materialsById.put(template.getId(), template);
            if (template.getSkills() != null)
                skillIds.addAll(extract(template.getSkills(), on(MaterialSkill.class).getId()));
        }

        materialTemplates.clear();
        materialTemplates = null;
    }

    public MaterialTemplate getTemplate(int materialId) {
        return materialsById.get(materialId);
    }

    public boolean isMaterialSkill(int skillId) {
        return skillIds.contains(skillId);
    }

    public int size() {
        return materialsById.size();
    }

}
